﻿/************************************************************************

   Extended Silverlight Toolkit

   Copyright (C) 2010-2012 Xceed Software Inc.

   This program is provided to you under the terms of the Microsoft Public
   License (Ms-PL) as published at http://extendedsilverlight.codeplex.com/license

   Please purchase a commercial version of this toolkit if you are using
   it in a commercial product. The commercial versions support the project,
   and also include more features for each control, priority support,
   source code, updates, removed watermark, and a proprietary license.

   Visit http://xceed.com and follow @datagrid on Twitter.

  **********************************************************************/

using System.Collections.Generic;

namespace Xceed.Silverlight.DataGrid
{
  internal class ColumnElementHostExtractorVisitor : VisitorBase
  {

    public ColumnElementHostExtractorVisitor()
      : base( ChildrenPolicy.AllChildren, true )
    {
    }

    public IEnumerable<IColumnElementHost> Hosts
    {
      get
      {
        return m_hosts;
      }
    }

    public override void Visit( Row row )
    {
      m_hosts.Add( row );
    }

    public override void Visit( RootHost rootHost )
    {
      base.Visit( rootHost );
      m_hosts.Add( rootHost );
    }

    private List<IColumnElementHost> m_hosts = new List<IColumnElementHost>();
  }
}
